package com.shine.demo.leetCode.leetCode1029;

import com.shine.demo.leetCode.entity.ListNode;

/**
 * 链表中倒数第k个节点
 * 输入一个链表，输出该链表中倒数第k个节点。为了符合大多数人的习惯，本题从1开始计数，即链表的尾节点是倒数第1个节点。
 * <p>
 * 例如，一个链表有 6 个节点，从头节点开始，它们的值依次是 1、2、3、4、5、6。这个链表的倒数第 3 个节点是值为 4 的节点。
 */
public class GetKthFromEnd {
    public static void main(String[] args) {
        ListNode l1 = new ListNode(1);
        ListNode l2 = new ListNode(2);
        ListNode l3 = new ListNode(3);
        ListNode l4 = new ListNode(4);
        ListNode l5 = new ListNode(5);
        l1.next = l2;
        l2.next = l3;
        l3.next = l4;
        l4.next = l5;
        System.out.println(getKthFromEnd(l1, 3));
    }

    public static ListNode getKthFromEnd(ListNode head, int k) {
        ListNode frontNode = head;
        ListNode behindNode = head;
        while (frontNode != null && k > 0) {
            //快指针先走K步
            frontNode = frontNode.next;
            k--;
        }
        while (frontNode != null) {
            //快指针走完，剩下的就是应该返回的
            frontNode = frontNode.next;
            behindNode = behindNode.next;
        }
        return behindNode;
    }

}
